package com.pdf;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Calendar;

import com.itextpdf.text.DocumentException;
import com.itextpdf.text.Element;
import com.itextpdf.text.pdf.BaseFont;
import com.itextpdf.text.pdf.PdfContentByte;
import com.itextpdf.text.pdf.PdfGState;
import com.itextpdf.text.pdf.PdfReader;
import com.itextpdf.text.pdf.PdfStamper;
 

public class padfTest {
	public static void main(String[] args) throws DocumentException, IOException {
		// 要输出的pdf文件
		BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(new File("E:/abc.pdf")));
		Calendar cal = Calendar.getInstance();
		SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd hh:mm:ss");
		// 将pdf文件先加水印然后输出
		setWatermark(bos, "E:/pdf源文件.pdf", format.format(cal.getTime()) + "  下载使用人：" + "测试user", 16);
	}

	public static void setWatermark(BufferedOutputStream bos, String input, String waterMarkName, int permission)
			throws DocumentException, IOException {
		PdfReader reader = new PdfReader(input);
		PdfStamper stamper = new PdfStamper(reader, bos);
		int total = reader.getNumberOfPages() + 1;
		PdfContentByte content;
		BaseFont base = BaseFont.createFont("STSong-Light", "UniGB-UCS2-H", BaseFont.EMBEDDED);
		PdfGState gs = new PdfGState();
		for (int i = 1; i < total; i++) {
			content = stamper.getOverContent(i);// 在内容上方加水印
			// content = stamper.getUnderContent(i);//在内容下方加水印
			gs.setFillOpacity(0.2f);
			// content.setGState(gs);
			content.beginText();
		//	content.setColorFill(Color.LIGHT_GRAY);
			content.setFontAndSize(base, 50);
			content.setTextMatrix(70, 200);
			content.showTextAligned(Element.ALIGN_CENTER, "公司内部文件，请注意保密！", 300, 350, 55);
		//	content.setColorFill(Color.BLACK);
			content.setFontAndSize(base, 8);
			content.showTextAligned(Element.ALIGN_CENTER, "下载时间：" + waterMarkName + "", 300, 10, 0);
			content.endText();

		}
		stamper.close();
	}
	
	public static void setWatermark1(BufferedOutputStream bos, String input, String waterMarkName, int permission)
			throws DocumentException, IOException {
//		byte[] buffer = new byte[10 * 1024];
//		
//		BufferedOutputStream bos1 = new BufferedOutputStream(new FileOutputStream(new File("E:/abc.pdf")));
//		
//			File file = new File();
//		    OutputStream output = new FileOutputStream(file);
//		    BufferedOutputStream bufferedOutput = new BufferedOutputStream(output);
//		    bufferedOutput.write(buffer);
		
		
		
		PdfReader reader = new PdfReader(input);
		PdfStamper stamper = new PdfStamper(reader, bos);
		
		//PdfStamper stamper = new PdfStamper
		int total = reader.getNumberOfPages() + 1;
		PdfContentByte content;
		BaseFont base = BaseFont.createFont("STSong-Light", "UniGB-UCS2-H", BaseFont.EMBEDDED);
		PdfGState gs = new PdfGState();
		for (int i = 1; i < total; i++) {
			content = stamper.getOverContent(i);// 在内容上方加水印
			// content = stamper.getUnderContent(i);//在内容下方加水印
			gs.setFillOpacity(0.2f);
			// content.setGState(gs);
			content.beginText();
		//	content.setColorFill(Color.LIGHT_GRAY);
			content.setFontAndSize(base, 50);
			content.setTextMatrix(70, 200);
			content.showTextAligned(Element.ALIGN_CENTER, "公司内部文件，请注意保密！", 300, 350, 55);
		//	content.setColorFill(Color.BLACK);
			content.setFontAndSize(base, 8);
			content.showTextAligned(Element.ALIGN_CENTER, "下载时间：" + waterMarkName + "", 300, 10, 0);
			content.endText();

		}
		stamper.close();
	}
	
	
}